import java.util.HashMap;
import java.util.Map;

public class SubarraySumsDivisibleByK {
    // 和可被k整除的子数组 https://leetcode.cn/problems/subarray-sums-divisible-by-k/description/
    public int subarraysDivByK(int[] nums, int k) {
        Map<Integer,Integer> hash=new HashMap<Integer,Integer>();
        hash.put(0%k,1);

        int sum=0;// 前一个位置的前缀和
        int ret=0;//返回结果
        for(int x:nums) {
            sum+=x;//此时的前缀和
            int r=(sum%k+k)%k;
            ret+=hash.getOrDefault(r,0);
            hash.put(r,hash.getOrDefault(r,0)+1);
        }
        return ret;
    }
}
